package com.nanfangzhe.shirodemo.config;

import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;

import org.apache.shiro.mgt.SecurityManager;
import org.apache.shiro.spring.web.ShiroFilterFactoryBean;
import org.apache.shiro.web.mgt.DefaultWebSecurityManager;
import org.apache.shiro.web.session.mgt.DefaultWebSessionManager;
import org.crazycake.shiro.RedisCacheManager;
import org.crazycake.shiro.RedisManager;
import org.crazycake.shiro.RedisSessionDAO;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import com.nanfangzhe.shirodemo.entity.UPermission;
import com.nanfangzhe.shirodemo.service.UPermissionService;

import redis.clients.jedis.JedisPool;
import redis.clients.jedis.JedisPoolConfig;

@Configuration
public class ShiroConfig {
	@Autowired
	UPermissionService permissionService;

	@Value("${spring.redis.host}")
	private String host;

	@Value("${spring.redis.port}")
	private int port;

	@Value("${spring.redis.database}")
	private int database;

	@Value("${spring.redis.password}")
	private String password;

	@Value("${spring.redis.timeout}")
	private int timeout;

	@Value("${spring.redis.jedis.pool.max-idle}")
	private int maxIdle;

	@Value("${spring.redis.jedis.pool.max-wait}")
	private long maxWaitMillis;

	/**
	 * ShiroFilterFactoryBean 处理拦截资源文件问题。
	 * 注意：单独一个ShiroFilterFactoryBean配置是或报错的，以为在
	 * 初始化ShiroFilterFactoryBean的时候需要注入：SecurityManager
	 *
	 * Filter Chain定义说明 1、一个URL可以配置多个Filter，使用逗号分隔 2、当设置多个过滤器时，全部验证通过，才视为通过
	 * 3、部分过滤器可指定参数，如perms，roles
	 *
	 */
	@Bean
	public ShiroFilterFactoryBean shirFilter(SecurityManager securityManager) {
		ShiroFilterFactoryBean bean = new ShiroFilterFactoryBean();

		// 必须设置 SecurityManager
		bean.setSecurityManager(securityManager);

		// 设置登录的请求
		bean.setLoginUrl("/home/no_login");
		// 登录成功后要跳转的链接
		bean.setSuccessUrl("/home/back");

		// 未授权跳转
		bean.setUnauthorizedUrl("/home/noauthor");

		// 拦截器. LinkedHashMap：有顺序的map
		Map<String, String> map = new LinkedHashMap<String, String>();
		// 配置不会被拦截的链接 顺序判断
		// filterChainDefinitionMap.put("/static/**", "anon");
		// filterChainDefinitionMap.put("/ajaxLogin", "anon");
		// 设置登录的请求
		map.put("/home/login", "anon");
		// 配置退出过滤器,其中的具体的退出代码Shiro已经替我们实现了
		map.put("/logout", "logout");

		List<UPermission> perms = permissionService.getAll();
		perms.forEach(perm -> {
			map.put(perm.getUrl(), perm.getName());
		});

		// filterChainDefinitionMap.put("/add", "perms[权限添加]");

		// <!-- 过滤链定义，从上向下顺序执行，一般将 /**放在最为下边 -->:这是一个坑呢，一不小心代码就不好使了;
		// <!-- authc:所有url都必须认证通过才可以访问; anon:所有url都都可以匿名访问-->
		// filterChainDefinitionMap.put("/**", "authc");

		bean.setFilterChainDefinitionMap(map);
		System.out.println("Shiro拦截器工厂类注入成功");
		return bean;
	}

	@Bean
	public SecurityManager securityManager() {
		// 默认安全管理器
		DefaultWebSecurityManager securityManager = new DefaultWebSecurityManager();
		// 设置realm.
		securityManager.setRealm(myShiroRealm());
		// 自定义缓存实现 使用redis
		securityManager.setCacheManager(cacheManager());
		// 自定义session管理 使用redis
		securityManager.setSessionManager(SessionManager());
		return securityManager;
	}

	/**
	 * 身份认证realm; (这个需要自己写，账号密码校验；权限等)
	 * 
	 * @return
	 */
	@Bean
	public MyShiroRealm myShiroRealm() {
		MyShiroRealm myShiroRealm = new MyShiroRealm();
		return myShiroRealm;
	}

	/**
	 * 配置shiro redisManager
	 * 
	 * @return
	 */
	/**
	 * @Title: redisManager @Description: TODO(描述这个方法的作用) @author:
	 * nanfangzhe @date: 2020年11月27日 上午10:06:05 @param: @return @return:
	 * RedisManager @throws
	 */
	public RedisManager redisManager() {
		RedisManager redisManager = new RedisManager();
		JedisPoolConfig jedisPoolConfig = new JedisPoolConfig();
		jedisPoolConfig.setMaxIdle(maxIdle);
		jedisPoolConfig.setMaxWaitMillis(maxWaitMillis);
		redisManager.setJedisPool(new JedisPool(jedisPoolConfig, host, port, timeout, password, database));
		return redisManager;
	}

	/**
	 * cacheManager 缓存 redis实现
	 * 
	 * @return
	 */
	public RedisCacheManager cacheManager() {
		RedisCacheManager redisCacheManager = new RedisCacheManager();
		redisCacheManager.setRedisManager(redisManager());
		// 过期时间30分钟
		redisCacheManager.setExpire(30 * 60);
		return redisCacheManager;
	}

	/**
	 * RedisSessionDAO shiro sessionDao层的实现 通过redis
	 */
	public RedisSessionDAO redisSessionDAO() {
		RedisSessionDAO redisSessionDAO = new RedisSessionDAO();
		redisSessionDAO.setRedisManager(redisManager());
		// 过期时间 设置为30分钟
		redisSessionDAO.setExpire(30 * 60);
		return redisSessionDAO;
	}

	/**
	 * shiro session的管理
	 */
	public DefaultWebSessionManager SessionManager() {
		DefaultWebSessionManager sessionManager = new DefaultWebSessionManager();
		sessionManager.setSessionDAO(redisSessionDAO());
		return sessionManager;
	}
}
